from langchain.agents import initialize_agent, AgentType
from common import create_calc_tools, llm


# 智能体初始化
agent = initialize_agent(
    tools=create_calc_tools(),  # 工具方法
    llm=llm,  # 大语言模型实例
    agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,  # agent类型
    verbose=True  # 打印调试信息
)
# 智能体的调用
resp = agent.invoke({"input": "深圳的天气怎么样?"})
print(resp)
